base_dir=/home/user4/workspace
dataset_name=$base_dir/ddg-benchmark/data

# esm_if1_gvp4_t16_142M_UR50 mifst mif
# esm_if1_gvp4_t16_142M_UR50.ef mifst.ef mif.ef
# carp_640M
# prot_t5_xl_uniref50 prot_t5_xl_bfd prot_bert prot_bert_bfd
# Tranception_Large Tranception_Medium Tranception_Small
# progen2_base progen2_BFD90 progen2_large progen2_medium progen2_oas progen2_small progen2_xlarge
# esm2_t6_8M_UR50D esm2_t12_35M_UR50D esm2_t30_150M_UR50D esm2_t33_650M_UR50D esm2_t36_3B_UR50D esm2_t48_15B_UR50D esm1v_t33_650M_UR90S_1 esm1v_t33_650M_UR90S_2 esm1v_t33_650M_UR90S_3 esm1v_t33_650M_UR90S_4 esm1v_t33_650M_UR90S_5 esm1b_t33_650M_UR50S
# RITA_s RITA_m RITA_l RITA_xl

python src/metric/spearmanr.py \
    --dataset_input $dataset_name \
    --mutant_site 0 \
    --model_location esm_if1_gvp4_t16_142M_UR50.ef mifst.ef mif.ef \
    --output_path ddg_ef.csv